Revista do CD-ROM 101/CD-ROM 101.iso/compl/maya5ple/Install_MayaPLE5_English.exe/Maya/Data1.cab/AEswitchControlTemplate.mel

Text File | 2003-07-17 | 13.4 KB | 447 lines

// Copyright (C) 1997-2002 Alias|Wavefront,
// a division of Silicon Graphics Limited.
//
// The information in this file is provided for the exclusive use of the
// licensees of Alias|Wavefront.  Such users have the right to use, modify,
// and incorporate this code into other products for purposes authorized
// by the Alias|Wavefront license agreement, without fee.
//
// ALIAS|WAVEFRONT DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
// INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO
// EVENT SHALL ALIAS|WAVEFRONT BE LIABLE FOR ANY SPECIAL, INDIRECT OR
// CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE,
// DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
// TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
//
//
//  Alias|Wavefront Script File
//
//  Creation Date:    March 6, 1998
//  Author:        cdt
//
//  Procedure Name:
//        AEswitchControlTemplate
//
//  Description Name;
//        Creates the custom attribute controls for the switch node.
//
//  Input Value:
//        nodeName        (currently unused)
//      inputAttr
//      shapeAttr
//      shadeAttr
//
//  Output Value:
//        None
//

proc string switchPlug( string $switchNode,
                        string $inputAttr,
                        string $inputIndex,
                        string $compoundAttr )
{
    return ($switchNode+"."+$inputAttr+"["+$inputIndex+"]."+$compoundAttr);
}

proc addLast( string $itemArray[], string $item )
{
    $itemArray[size($itemArray)] = $item;
}

proc string plugAttr( string $plug )
{
    string $buffer[];
    tokenize($plug, ".", $buffer);
    return $buffer[1];
}

proc downStreamNetwork( string $node, string $result[] )
{
    if (`objectType $node` == "shadingEngine") {
        // Check if the node is already in the result. This may occur if
        // the outputs from a node branch and feed back into the same
        // shading engine. For example, a switch node connected to a
        // placement.

        int $isFound = false;

        for ($item in $result) {
            if ($item == $node) {
                $isFound = true;
                break;
            }
        }

        if (!$isFound) addLast($result, $node);
    }
    else {
        string $connections[] = `listConnections -connections true
            -source false -destination true $node`;

        // Foreach destination (output) connection on node.
        for ($i = 0; $i < size($connections); $i += 2) {
            string $srcPlug = $connections[$i];
            string $dstNode = $connections[$i + 1];

            // Is it not a message connection ?
            if (plugAttr($srcPlug) != "message") {
                // We follow the first non-message output connection.
                downStreamNetwork($dstNode, $result);
            }
        }
    }
}

//
//  Procedure Name:
//      AEswitchRemoveCB
//
//  Description:
//      Procedure to remove a row from the switch table.
//
global proc AEswitchRemoveCB( string $shapePlug,
                              string $shadePlug)
{
    string $shapeCons[] = `listConnections -plugs true $shapePlug`;

    for ($item in $shapeCons) {
        disconnectAttr $item $shapePlug;
    }

    string $shadeCons[] = `listConnections -plugs true $shadePlug`;

    for ($item in $shadeCons) {
        disconnectAttr $item $shadePlug;
    }
}

//
//  Procedure Name:
//      AEswitchMenuCB
//
//  Description:
//      Procedure to build a popup menu specific to the spread sheet
//      row the mouse is above. This menu will be displayed when the
//      user presses the RMB.
//
global proc AEswitchMenuCB( string $parent,
                            string $menuName,
                            string $inputAttr,
                            string $shapeAttr,
                            string $shadeAttr )
{
    setParent $parent;

    setParent -m $menuName;

    popupMenu -e -deleteAllItems $menuName;

    string $switchNode = `switchTable -q -switchNode switchTable`;

    int $row = `switchTable -q -underPointerRow switchTable`;

    if ($row == -1) return;

    // Create shape and shade plugs for the element under the pointer.
    //

    string $shapePlug = switchPlug($switchNode, $inputAttr, $row, $shapeAttr);
    string $shadePlug = switchPlug($switchNode, $inputAttr, $row, $shadeAttr);

    // Create menu item to show the AE of the nodes connected to shape plug
    //

    string $shapeCons[] = `listConnections $shapePlug`;

    for ($item in $shapeCons) {
        menuItem -l ("Edit "+$item+" ...") -c ("showEditor "+$item);
    }

    // Create menu item to show the AE of the nodes connected to shade plug
    //

    string $shadeCons[] = `listConnections -plugs true $shadePlug`;

    for ($item in $shadeCons) {
        menuItem -l ("Edit "+$item+" ...") -c ("showEditor "+$item);
    }

    // Create a menu item to map a new shader to the shade plug
    //

    menuItem -l "Map ..."
        -c ("createRenderNode -allWithTexturesUp "+
            ("\"defaultNavigation -connectToExisting -source %node "+
             "-destination "+$shadePlug+"\" ")+
            "\"\"");

    // Create a menu item to disconnect the shape and shade plugs.
    //

    menuItem -l "Remove item"
        -c ("AEswitchRemoveCB "+$shapePlug+" "+$shadePlug);
}

//
//  Procedure Name:
//      AEswitchAddButtonCB
//
//  Description:
//      Procedure to connect all surfaces downstream of the
//      switch. This be invoked from the add button in the AE.
//
global proc AEswitchAddButtonCB( string $switchNode,
                                 string $inputAttr,
                                 string $shapeAttr,
                                 string $shadeAttr )
{
    int $nextAvailable = 0;

    string $currentSurfaces[];

    // Create the current list of surfaces. We filter switchConnections
    // because it will contain connections to both the shape and shade.
    //

    string $switchConnections[] = `listConnections -connections true
        -plugs true ($switchNode+"."+$inputAttr)`;

    for ($i = 0; $i < size($switchConnections); $i += 2) {
        // Is this a shape connection ?
        if (match("\\."+$shapeAttr+"$", $switchConnections[$i]) != "") {
            addLast($currentSurfaces, $switchConnections[$i+1]);
        }
    }

    // Search downstream to find all shading groups
    //

    string $shadingGroups[];
    downStreamNetwork( $switchNode, $shadingGroups );

    if (size($shadingGroups) == 0) {
        error -showLineNumber false
            ($switchNode+" is not connected to a shading group,"+
             " unable to add surfaces");
    }

    // Foreach shading group add the surfaces not already connected
    //

    int $wereSurfacesAdded = false;

    for ($group in $shadingGroups) {
        string $surfaces[] =
            `listConnections -plugs true ($group+".dagSetMembers")`;

        for ($surface in $surfaces) {
            // Ensure the surface is not already connected
            //

            int $isFound = false;

            for ($currentSurface in $currentSurfaces) {
                if ($surface == $currentSurface) {
                    $isFound = true;
                    break;
                }
            }

            if (!$isFound) {
                $wereSurfacesAdded = true;

                // Find the next available connection slot
                //

                string $shapePlug, $shadePlug;

                while (true) {
                    $shapePlug = switchPlug( $switchNode, $inputAttr,
                                             $nextAvailable, $shapeAttr );

                    $shadePlug = switchPlug( $switchNode, $inputAttr,
                                             $nextAvailable, $shadeAttr );

                    string $shape[] = `listConnections $shapePlug`;

                    string $shade[] = `listConnections $shadePlug`;

                    // Is there no connections to the shape and shade ?
                    if (size($shape) == 0 && size($shade) == 0)
                        break;

                    $nextAvailable += 1;
                }

                connectAttr $surface $shapePlug;

                // The same surface cannot be in two shading groups,
                // but it doesn't hurt to ensure we don't add it
                // again.

                addLast($currentSurfaces, $surface);
            }
        }
    }

    if (!$wereSurfacesAdded) {
        warning -showLineNumber false
            ("There are no more surfaces to add to "+$switchNode+","+
             " they have already been added");
    }
}

//
//  Procedure Name:
//      AEswitchRemoveButtonCB
//
//  Description:
//      Procedure to break connections to the shape and shade of the
//      selected switch table row. This will be invoked from the
//      remove button in the AE.
//
global proc AEswitchRemoveButtonCB( string $switchNode,
                                    string $inputAttr,
                                    string $shapeAttr,
                                    string $shadeAttr )
{
    int $row = `switchTable -q -selectedRow switchTable`;

    if ($row == -1) return;

    string $shapePlug = switchPlug($switchNode, $inputAttr, $row, $shapeAttr);

    string $shadePlug = switchPlug($switchNode, $inputAttr, $row, $shadeAttr);

    AEswitchRemoveCB($shapePlug, $shadePlug);
}

//
//  Procedure Name:
//      AEswitchMapButtonCB
//
//  Description:
//      Procedure to create a new shader and connect it to the
//      selected switch table row. This will be invoked from the map
//      button in the AE.
//
global proc AEswitchMapButtonCB( string $switchNode,
                                 string $inputAttr,
                                 string $shapeAttr,
                                 string $shadeAttr )
{
    int $row = `switchTable -q -selectedRow switchTable`;

    if ($row == -1) return;

    string $shadePlug = switchPlug($switchNode, $inputAttr, $row, $shadeAttr);

    createRenderNode -allWithTexturesUp
        ("defaultNavigation -connectToExisting -source %node -destination "+
         $shadePlug) "";
}

//
//  Procedure Name:
//      AEswitchControlNew
//
//  Description:
//      Procedure to create the switch table for a switch node.
//
global proc AEswitchControlNew( string $inputAttr,
                                string $shapeAttr,
                                string $shadeAttr,
                                string $messagePlug )
{
    global int $gTextColumnWidthIndex;
    global int $gAESingleWidgetWidthIndex;

    separator -style "in";

    int    $switchButtonSize = ((3 * $gAESingleWidgetWidthIndex) / 2 );

    rowLayout -numberOfColumns 3
        -cw 1 $gTextColumnWidthIndex
        -cw 2 $switchButtonSize
        -cw 3 $switchButtonSize

        -cal 1 "center" -cal 2 "center" -cal 3 "center"
        -cat 1 "right" 1 -cat 2 "both" 1 -cat 3 "both" 1;
        button -l "Add Surfaces" -w $switchButtonSize addButton;
        button -l "Remove Item" removeButton;
        button -l "Map Item" mapButton;
        setParent ..;

    separator -style "none";

    formLayout switchForm;
        switchTable switchTable;
        popupMenu -button 3 switchMenu;
        setParent ..;

    formLayout -e
        -height 150
        -af switchTable top 0
        -af switchTable left ($gTextColumnWidthIndex - $switchButtonSize)
        -af switchTable bottom 0
        -af switchTable right 0
        switchForm;

    separator -style "in";

    AEswitchControlReplace($inputAttr, $shapeAttr, $shadeAttr, $messagePlug);
}

//
//  Procedure Name:
//      AEswitchControlNew
//
//  Description:
//      Procedure to initialize the switch table for a node.
//
global proc AEswitchControlReplace( string $inputAttr,
                                    string $shapeAttr,
                                    string $shadeAttr,
                                    string $messagePlug )
{
    string $buffer[];
    tokenize($messagePlug, ".", $buffer);
    string $nodeName = $buffer[0];

    switchTable -e
        -label1 $shapeAttr -label2 $shadeAttr
        -switchNode $nodeName
        switchTable;

    string $parent = `setParent -q`;
    string $menuName = $parent + "|switchForm|switchTable|switchMenu";
    popupMenu -e -postMenuCommand
        ("AEswitchMenuCB "
         +$parent+" "+$menuName+" "+$inputAttr+" "+$shapeAttr+" "+$shadeAttr)
        switchMenu;

    button -e -command
        ("AEswitchAddButtonCB "+$nodeName+" "+ $inputAttr+" "+
         $shapeAttr+" "+$shadeAttr) addButton;

    button -e -command
        ("AEswitchRemoveButtonCB "+$nodeName+" "+$inputAttr+" "+
         $shapeAttr+" "+ $shadeAttr) removeButton;

    button -e -command
        ("AEswitchMapButtonCB "+$nodeName+" "+$inputAttr+" "+
         $shapeAttr+" "+ $shadeAttr) mapButton;
}

global proc AEswitchControlTemplate( string $nodeName,
                                     string $inputAttr,
                                     string $shapeAttr,
                                     string $shadeAttr)
{
    editorTemplate -callCustom
        ("AEswitchControlNew "+$inputAttr+" "+$shapeAttr+" "+$shadeAttr)
        ("AEswitchControlReplace "+$inputAttr+" "+$shapeAttr+" "+$shadeAttr)
        "message";
}